题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的J前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
解题思路
前序遍历的顺序为根->左->右,所以前序遍历第一个数字是根节点,中序遍历的顺序为左->根->右,在数组中找到根节点可以分出左子树和右子树,所以
分割后的中序遍历
左子树为{4,7,2},右子树为{5,3,8,6},根节点为1
前序遍历
左子树为{2,4,7},右子树为{3,5,6,8},根节点为1
代码如下
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28
|
import java.util.Arrays; public class Solution { public TreeNode reConstructBinaryTree(int [] pre,int [] in) { if(pre.length==0|in.length==0){ return null; } TreeNode root = new TreeNode(pre[0]); for(int i =0;i<in.length;i++){ if(in[i]==pre[0]){ root.left=reConstructBinaryTree(Arrays.copyOfRange(pre,1,i+1),Arrays.copyOfRange(in,0,i)); root.right=reConstructBinaryTree(Arrays.copyOfRange(pre,i+1,pre.length),Arrays.copyOfRange(in,i+1,in.length)); break; } }return root; } }
|
盲点
对java的类不熟悉,TreeNode和Arrays的方法不熟悉,加强学习